package com.yunxi.module.infra.service.file.impl;

import cn.hutool.core.io.resource.ResourceUtil;
import cn.hutool.core.util.IdUtil;
import com.google.common.cache.CacheLoader;
import com.google.common.cache.LoadingCache;
import com.yunxi.framework.common.exception.util.ServiceExceptionUtil;
import com.yunxi.framework.common.pojo.PageResult;
import com.yunxi.framework.common.util.cache.CacheUtils;
import com.yunxi.framework.common.util.json.JsonUtils;
import com.yunxi.framework.common.util.validation.ValidationUtils;
import com.yunxi.module.infra.convert.file.FileConfigConvert;
import com.yunxi.module.infra.enums.ErrorCodeConstants;
import com.yunxi.module.infra.framework.file.core.client.FileClient;
import com.yunxi.module.infra.framework.file.core.client.FileClientConfig;
import com.yunxi.module.infra.framework.file.core.client.factory.FileClientFactory;
import com.yunxi.module.infra.framework.file.core.enums.FileStorageEnum;
import com.yunxi.module.infra.mapper.file.FileConfigMapper;
import com.yunxi.module.infra.model.file.FileConfigDO;
import com.yunxi.module.infra.service.file.FileConfigService;
import com.yunxi.module.infra.vo.admin.file.config.FileConfigPageReqVO;
import com.yunxi.module.infra.vo.admin.file.config.FileConfigSaveReqVO;
import lombok.Getter;
import lombok.extern.slf4j.Slf4j;
import org.simpleframework.xml.core.Validate;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import javax.validation.Validator;
import java.time.Duration;
import java.util.Map;
import java.util.Objects;

/**
 * <p>
 * 文件配置 Service 接口实现类
 * </p>
 *
 * @author lidy
 * @since 2024-05-18
 */
@Service
@Validate
@Slf4j
public class FileConfigServiceImpl implements FileConfigService {

    private static final Long CACHE_MASTER_ID = 0L;

    @Resource
    private FileConfigMapper fileConfigMapper;

    @Resource
    private FileClientFactory fileClientFactory;

    /**
     * {@link FileClient} 缓存，通过它异步刷新 fileClientFactory
     */
    @Getter
    private final LoadingCache<Long, FileClient> clientCache = CacheUtils.buildAsyncReloadingCache(Duration.ofSeconds(10L),
            new CacheLoader<Long, FileClient>() {

                @Override
                public FileClient load(Long id) {
                    FileConfigDO config = Objects.equals(CACHE_MASTER_ID, id) ?
                            fileConfigMapper.selectByMaster() : fileConfigMapper.selectById(id);
                    if (config != null) {
                        fileClientFactory.createOrUpdateFileClient(config.getId(), config.getStorage(), config.getConfig());
                    }
                    return fileClientFactory.getFileClient(null == config ? id : config.getId());
                }

            });

    @Resource
    private Validator validator;


    /**
     * 获得 master 文件客户端
     *
     * @return 文件客户端
     */
    @Override
    public FileClient getMasterFileClient() {

        return clientCache.getUnchecked(CACHE_MASTER_ID);
    }

    /**
     * 获得文件配置分页列表
     *
     * @param pageReqVO 分页查询参数
     * @return 文件配置分页列表
     */
    @Override
    public PageResult<FileConfigDO> getFileConfigPage(FileConfigPageReqVO pageReqVO) {
        return fileConfigMapper.selectPage(pageReqVO);
    }

    /**
     * 测试文件配置是否正确
     *
     * @param configId 文件配置 ID
     * @return 文件 URL
     */
    @Override
    public String testFileConfig(Long configId) throws Exception {
        // 根据配置 ID 校验配置是否存在
        validateFileConfigExists(configId);
        // 上传文件
        byte[] content = ResourceUtil.readBytes("file/avatar_test.jpg");

        return getFileClient(configId).upload(content, IdUtil.fastSimpleUUID() + ".jpg", "image/jpeg");
    }

    /**
     * 根据配置 ID 获得配置
     *
     * @param configId 配置 ID
     * @return 配置信息
     */
    @Override
    public FileClient getFileClient(Long configId) {
        return clientCache.getUnchecked(configId);
    }

    /**
     * 获得文件配置
     *
     * @param configId 配置 ID
     * @return 配置信息
     */
    @Override
    public FileConfigDO getFileConfig(Long configId) {
        return fileConfigMapper.selectById(configId);
    }

    /**
     * 删除文件配置
     *
     * @param configId 配置编号
     */
    @Override
    public void deleteFileConfig(Long configId) {
        // 校验是否存在
        FileConfigDO config = validateFileConfigExists(configId);
        if (Boolean.TRUE.equals(config.getMaster())) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.FILE_CONFIG_DELETE_FAIL_MASTER);
        }
        // 删除
        fileConfigMapper.deleteById(configId);

        // 清空缓存
        clearCache(configId, null);
    }

    /**
     * 更新文件配置
     *
     * @param updateReqVO 文件配置信息
     */
    @Override
    public void updateFileConfig(FileConfigSaveReqVO updateReqVO) {
        // 校验是否存在
        FileConfigDO config = validateFileConfigExists(updateReqVO.getId());

        // 更新
        // 设置更新信息
        FileConfigDO fileConfigDO = FileConfigConvert.INSTANCE.convert(updateReqVO)
                .setConfig(parseClientConfig(config.getStorage(), updateReqVO.getConfig()));
        fileConfigMapper.updateById(fileConfigDO);

        // 清空缓存
        clearCache(config.getId(), null);
    }

    /**
     * 创建文件配置
     *
     * @param createReqVO 文件配置信息
     * @return 配置编号
     */
    @Override
    public Long createFileConfig(FileConfigSaveReqVO createReqVO) {
        // 创建文件配置信息
        FileConfigDO fileConfigDO = FileConfigConvert.INSTANCE.convert(createReqVO)
                .setConfig(parseClientConfig(createReqVO.getStorage(), createReqVO.getConfig()))
                .setMaster(false);
        fileConfigMapper.insert(fileConfigDO);
        return fileConfigDO.getId();
    }

    /**
     * 更新文件配置为主配置
     *
     * @param configId 配置编号
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void updateFileConfigMaster(Long configId) {
        // 校验配置是否存在
        validateFileConfigExists(configId);

        // 更新其他为非主配置
        fileConfigMapper.updateBatch(new FileConfigDO().setMaster(false));

        // 更新
        fileConfigMapper.updateById(new FileConfigDO().setId(configId).setMaster(true));

        // 清空缓存
        clearCache(null, true);
    }

    /**
     * 转换文件存储配置
     *
     * @param storage 存储器
     * @param config  存储配置
     * @return 文件配置
     */
    private FileClientConfig parseClientConfig(Integer storage, Map<String, Object> config) {
        // 获得配置类
        Class<? extends FileClientConfig> configClass = FileStorageEnum.getByStorage(storage).getConfigClass();
        // 转换成文件配置
        FileClientConfig clientConfig = JsonUtils.parseObject2(JsonUtils.toJsonString(config), configClass);
        // 参数校验
        ValidationUtils.validate(validator, clientConfig);
        // 设置参数
        return clientConfig;
    }

    /**
     * 清空指定的文件配置缓存
     *
     * @param configId 配置编号
     * @param master   是否主配置
     */
    private void clearCache(Long configId, Boolean master) {
        if (configId != null) {
            clientCache.invalidate(configId);
        }
        if (Boolean.TRUE.equals(master)) {
            clientCache.invalidate(CACHE_MASTER_ID);
        }
    }


    /**
     * 根据配置 ID 校验配置是否存在
     *
     * @param configId 配置 ID
     */
    private FileConfigDO validateFileConfigExists(Long configId) {
        FileConfigDO fileConfigDO = fileConfigMapper.selectById(configId);
        if (fileConfigDO == null) {
            throw ServiceExceptionUtil.exception(ErrorCodeConstants.FILE_CONFIG_NOT_EXISTS);
        }
        return fileConfigDO;
    }


}
